package com.copa.l1797;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

/**
 * @author copa
 * @createDate 2023-02-09 09:06
 * @function 常规法
 */
public class Test1 {

    /**
     * 验证码存活的时间
     */
    private int timeToLive;

    /**
     * key:tokenId value:过期时间
     */
    private Map<String, Integer> map;

    public Test1(int timeToLive) {
        this.timeToLive = timeToLive;
        this.map = new HashMap<>();
    }

    public void generate(String tokenId, int currentTime) {
        if (Objects.isNull(map.get(tokenId))) {
            // 如果不存在该tokenId，可以生成
            map.put(tokenId, currentTime + timeToLive);
        }
    }

    public void renew(String tokenId, int currentTime) {
        Integer expireTime = map.get(tokenId);
        if (Objects.nonNull(expireTime) && currentTime < expireTime) {
            // 如果存在该tokenId且小于当前时间，可以根据当前时间累加验证码存活的时间
            map.put(tokenId, currentTime + timeToLive);
        }
    }

    public int countUnexpiredTokens(int currentTime) {
        int nonExpiredTokenCount = 0;
        for (Integer expireTime : map.values()) {
            if (expireTime > currentTime) {
                // 统计若没有超过过期时间的就累加
                nonExpiredTokenCount++;
            }
        }
        return nonExpiredTokenCount;
    }

    public static void main(String[] args) {
        // 构造 Test1 ，设置 timeToLive = 5 秒。
        Test1 test1 = new Test1(5);
        // 时刻 1 时，没有验证码的 tokenId 为 "aaa" ，没有验证码被更新。
        test1.renew("aaa", 1);
        // 时刻 2 时，生成一个 tokenId 为 "aaa" 的新验证码。
        test1.generate("aaa", 2);
        // 时刻 6 时，只有 tokenId 为 "aaa" 的验证码未过期，所以返回 1 。
        System.out.println(test1.countUnexpiredTokens(6));
        // 时刻 7 时，生成一个 tokenId 为 "bbb" 的新验证码。
        test1.generate("bbb", 7);
        // tokenId 为 "aaa" 的验证码在时刻 7 过期，且 8 >= 7 ，所以时刻 8 的renew 操作被忽略，没有验证码被更新。
        test1.renew("aaa", 8);
        // tokenId 为 "bbb" 的验证码在时刻 10 没有过期，所以 renew 操作会执行，该 token 将在时刻 15 过期。
        test1.renew("bbb", 10);
        // tokenId 为 "bbb" 的验证码在时刻 15 过期，tokenId 为 "aaa" 的验证码在时刻 7 过期，所有验证码均已过期，所以返回 0 。
        System.out.println(test1.countUnexpiredTokens(15));
    }

}
